https://kukuruku.co/post/introduction-to-machine-learning-with-python-andscikit-learn/